package jnpf.model.ragFlow.know.api;

import io.swagger.annotations.ApiModelProperty;
import lombok.Data;

@Data
public class AssistantConfigResp {
    @ApiModelProperty(value = "id")
    private String id = "";

    @ApiModelProperty(value = "助理名称")
    private String name = "";


    @ApiModelProperty(value = "助理描述")
    private String description = "";

    @ApiModelProperty(value = "助手头像")
    private String icon = "";

    @ApiModelProperty(value = "语言")
    private String language = "English";

    @ApiModelProperty(value = "提示配置")
    private PromptConfig prompt_config = new PromptConfig();

    @ApiModelProperty(value = "知识库ID列表")
    private String[] kb_ids = new String[]{};

    @ApiModelProperty(value = "大语言模型ID")
    private String llm_id = "moonshot-v1-8k@Moonshot";

    @ApiModelProperty(value = "大语言模型设置")
    private LlmSetting llm_setting = new LlmSetting();

    @ApiModelProperty(value = "相似度阈值")
    private Double similarity_threshold = 0.2;

    @ApiModelProperty(value = "关键字相似度权重")
    private Double vector_similarity_weight = 0.3;

    @ApiModelProperty(value = "Top N")
    private Integer top_n = 8;

    @ApiModelProperty(value = "Rerank模型")
    private String rerank_id = "gte-rerank@Tongyi-Qianwen";

    @ApiModelProperty(value = "向量检索返回数量")
    private Integer top_k = 1024;


}
